from enum import Enum


class ModelDatabaseEnum(Enum):
    id = 'id'
    model_path = 'model_path'
    model_name = 'model_name'
    model_description = 'model_description'
    model_data_files = 'model_data_files'
    train_time = 'train_time'
    increasing_base_on = 'increasing_base_on'

    iterations = 'iterations'
    depth = 'depth'
    learning_rate = 'learning_rate'
    test_size = 'test_size'  # 测试集占比
    f1 = 'f1'
    recall = 'recall'
    accuracy = 'accuracy'
    precision = 'precision'
    sample_type = 'sample_type'

    eval_metric = 'eval_metric'
    use_best_model = 'use_best_model'
    cluster = 'cluster'
    is_grid_search = 'is_grid_search'
